package org.example.rocketmq.msg_type.affairs;

import org.apache.commons.lang3.ThreadUtils;
import org.apache.rocketmq.client.producer.*;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageQueue;
import org.apache.rocketmq.remoting.common.RemotingHelper;

import java.time.Duration;
import java.util.List;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.ThreadFactory;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * 事务消息
 */
public class ProducerAffairs {

    public static void main(String[] args) throws Exception {
        producer();
    }


    /**
     * 事务消息： 是 RocketMQ 提供的一个非常有特色的功能
     * <p>
     * 事务消息是在分布式系统中保证最终一致性的两阶段提交的消息实现。他可以保证本地事务执行与消息发送两个操作的原子性，也就是这两个操作一起成功或一起失败。
     * 事务消息机制的关键是在发送消息时会将消息转为一个 half半消息，并存入RocketMQ内部的一个Topic（RMQ_SYS_TRANS_HALF_TOPIC），
     * 这个 topic 对消费者是不可见的，再经过一系列事务检查通过后，再将消息转存到目标 topic ，这样对消费者就可见了
     * <p>
     * 事务消息只保证消息发送者的本地事务与发消息这两个操作的原子性，因此，事务消息的示例只涉及到消息发送者，对于消费者来说，并没有什么特别的。
     * 事务消息的关键是在 TransactionMQProducer 中指定了一个 TransactionListener 事务监听器，这个事务监听器就是事务消息的关键控制器。
     * <p>
     * 使用限制：
     * 1. 事务消息不支持延迟消息和批量消息
     * 2. 为了避免单个消息被检查太多次而导致半队列消息累计，我们默认将单个消息的检查次数限制为15次，但是用户可以通过 Broker 配置文件的 transactionCheckMax 参数来修改此限制。
     * 如果已经检查某条消息超过N次的话（N = transactionCheckMax）则 Broker 将丢弃次消息，并在默认情况下同时打印错误日志。可以通过重写 AbstractTransactionCheckListener类来修改这个行为
     * 3. 事务性消息可能不止一次被检查或消费
     */
    public static void producer() throws Exception {
        TransactionMQProducer producer = new TransactionMQProducer("producer_group_01");
        producer.setNamesrvAddr("rocketmq.jun.cn:9876");

        // 异步提交事务状态，提高性能
        new ThreadPoolExecutor(2, 5, 100, TimeUnit.SECONDS, new ArrayBlockingQueue<>(2000), new ThreadFactory() {
            @Override
            public Thread newThread(Runnable r) {
                Thread thread = new Thread(r);
                thread.setName("ExecutorService-baiLi");
                return thread;
            }
        });
        // 本地事务监听器
        producer.setTransactionListener(new TransactionListenerImpl());
        producer.start();

        String[] tags = new String[]{"TagA", "TagB", "TagC", "TagD", "TagE"};
        for (int i = 0; i < 10; i++) {
            String tagName = tags[i % tags.length];
            byte[] body = ("Hello RocketMQ " + i + "_step_" + tagName).getBytes(RemotingHelper.DEFAULT_CHARSET);
            Message message = new Message("TopicTest", tagName, body);

            TransactionSendResult result = producer.sendMessageInTransaction(message, null);

            System.out.printf("消息发送成功 %s%n", result);
            ThreadUtils.sleep(Duration.ofSeconds(1));

        }

        ThreadUtils.sleep(Duration.ofSeconds(10000));
        producer.shutdown();
    }

}
